#!/usr/bin/env sh
set -eu

# Downloads the latest tarball from https://zed.dev/releases and unpacks it
# into ~/.local/. If you'd prefer to do this manually, instructions are at
# https://zed.dev/docs/linux.

main() {
	platform="$(uname -s)"
	arch="$(uname -m)"
	channel="${ZED_CHANNEL:-stable}"
	temp="$(mktemp -d "/tmp/zed-XXXXXX")"

	if [ "$platform" = "Darwin" ]; then
		platform="macos"
	elif [ "$platform" = "Linux" ]; then
		platform="linux"
	else
		echo "Unsupported platform $platform"
		exit 1
	fi

	case "$platform-$arch" in
	macos-arm64* | linux-arm64* | linux-armhf | linux-aarch64)
		arch="aarch64"
		;;
	macos-x86* | linux-x86* | linux-i686*)
		arch="x86_64"
		;;
	*)
		echo "Unsupported platform or architecture"
		exit 1
		;;
	esac

	if which curl >/dev/null 2>&1; then
		curl() {
			command curl -fL "$@"
		}
	elif which wget >/dev/null 2>&1; then
		curl() {
			wget -O- "$@"
		}
	else
		echo "Could not find 'curl' or 'wget' in your path"
		exit 1
	fi

	"$platform" "$@"

	if [ "$(which "zed")" = "$HOME/.local/bin/zed" ]; then
		echo "Zed has been installed. Run with 'zed'"
	else
		echo "To run Zed from your terminal, you must add ~/.local/bin to your PATH"
		echo "Run:"

		case "$SHELL" in
		*zsh)
			echo "   echo 'export PATH=\$HOME/.local/bin:\$PATH' >> ~/.zshrc"
			echo "   source ~/.zshrc"
			;;
		*)
			echo "   echo 'export PATH=\$HOME/.local/bin:\$PATH' >> ~/.bashrc"
			echo "   source ~/.bashrc"
			;;
		esac

		echo "To run Zed now, '~/.local/bin/zed'"
	fi
}

linux() {
	if [ -n "${ZED_BUNDLE_PATH:-}" ]; then
		cp "$ZED_BUNDLE_PATH" "$temp/zed-linux-$arch.tar.gz"
	else

		# get current version number
		new_version=`curl -i -s https://zed.dev/download | awk '{match($0, /releases\/stable\/([0-9]+.[0-9]+.[0-9]+)"/);print substr($0, RSTART+16, RLENGTH-17)}' | uniq | sed -e '/^$/d'`
		echo "new version: $new_version";

		# get local version number		
		local_version=`echo $(zed --version) | awk '{match($0, /Zed ([0-9]+.[0-9]+.[0-9]+) /);print substr($0, RSTART+4, RLENGTH-5)}'`
		echo "local version: $local_version"

		# do need to update?
		has_new=`expr $new_version \> $local_version`
		# echo $hasNew;
		if [ $has_new -eq 1 ];then
		  echo "updating...";
		else
		  echo "No need to update, exiting soon.";
		  exit 0;
		fi		
		
		echo "Downloading Zed"
		# exit 0;
		# ZED_VERSION="0.187.5"
		ZED_VERSION=$new_version
		# zed_linux_download_url="https://zed.dev/api/releases/$channel/latest/zed-linux-$arch.tar.gz"
		zed_linux_download_url="https://gh.monlor.com/https://github.com/zed-industries/zed/releases/download/v${ZED_VERSION}/zed-linux-x86_64.tar.gz"
		echo $zed_linux_download_url
		curl "$zed_linux_download_url" >"$temp/zed-linux-$arch.tar.gz"
	fi

	suffix=""
	if [ "$channel" != "stable" ]; then
		suffix="-$channel"
	fi

	appid=""
	case "$channel" in
	stable)
		appid="dev.zed.Zed"
		;;
	nightly)
		appid="dev.zed.Zed-Nightly"
		;;
	preview)
		appid="dev.zed.Zed-Preview"
		;;
	dev)
		appid="dev.zed.Zed-Dev"
		;;
	*)
		echo "Unknown release channel: ${channel}. Using stable app ID."
		appid="dev.zed.Zed"
		;;
	esac

	# Unpack
	rm -rf "$HOME/.local/zed$suffix.app"
	mkdir -p "$HOME/.local/zed$suffix.app"
	tar -xzf "$temp/zed-linux-$arch.tar.gz" -C "$HOME/.local/"

	# Setup ~/.local directories
	mkdir -p "$HOME/.local/bin" "$HOME/.local/share/applications"

	# Link the binary
	if [ -f "$HOME/.local/zed$suffix.app/bin/zed" ]; then
		ln -sf "$HOME/.local/zed$suffix.app/bin/zed" "$HOME/.local/bin/zed"
	else
		# support for versions before 0.139.x.
		ln -sf "$HOME/.local/zed$suffix.app/bin/cli" "$HOME/.local/bin/zed"
	fi

	# Copy .desktop file
	desktop_file_path="$HOME/.local/share/applications/${appid}.desktop"
	cp "$HOME/.local/zed$suffix.app/share/applications/zed$suffix.desktop" "${desktop_file_path}"
	sed -i "s|Icon=zed|Icon=$HOME/.local/zed$suffix.app/share/icons/hicolor/512x512/apps/zed.png|g" "${desktop_file_path}"
	sed -i "s|Exec=zed|Exec=$HOME/.local/zed$suffix.app/libexec/zed-editor|g" "${desktop_file_path}"
}

macos() {
	echo "Downloading Zed"
	curl "https://zed.dev/api/releases/$channel/latest/Zed-$arch.dmg" >"$temp/Zed-$arch.dmg"
	hdiutil attach -quiet "$temp/Zed-$arch.dmg" -mountpoint "$temp/mount"
	app="$(
		cd "$temp/mount/"
		echo *.app
	)"
	echo "Installing $app"
	if [ -d "/Applications/$app" ]; then
		echo "Removing existing $app"
		rm -rf "/Applications/$app"
	fi
	ditto "$temp/mount/$app" "/Applications/$app"
	hdiutil detach -quiet "$temp/mount"

	mkdir -p "$HOME/.local/bin"
	# Link the binary
	ln -sf "/Applications/$app/Contents/MacOS/cli" "$HOME/.local/bin/zed"
}

main "$@"
